#include"../comm/httplib.h"
#include"oj_control.hpp"
#include<jsoncpp/json/json.h>
#include<signal.h>

using namespace httplib;
using namespace ns_control;

void MyDaemon()
{
    signal(SIGPIPE, SIG_IGN);
    signal(SIGCHLD, SIG_IGN);
    if(fork() > 0) exit(0);
    setsid();
    int fd = open("/dev/null", O_RDWR);
    if(fd > 0)
    {
        dup2(0, fd);
        dup2(1, fd);
        dup2(2, fd);
        close(fd);
    }
}

static Control* ctrl_ptr = nullptr;
void Recovery(int signo)
{
    ctrl_ptr->RecoverMachines();
}

int main()
{
    MyDaemon(); // 守护进程

    signal(SIGQUIT, Recovery); // 一键上线所有主机

    Server svr; //用户请求的服务路由功能

    Control ctrl;
    ctrl_ptr = &ctrl;
    
    // 获取所有的题目列表
    svr.Get("/all_questions", [&ctrl](const Request& req, Response& res){
        std::string html;
        ctrl.AllQuestions(&html);
        //用户看到的是什么呢？？网页数据 + 拼上了题目相关的数据
        res.set_content(html, "text/html; charset=utf-8");
    });
    // 用户要根据题目编号，获取题目的内容
    // /question/100 -> 正则匹配
    // R"()", 原始字符串raw string,保持字符串内容的原貌，不用做相关的转义
    svr.Get(R"(/one_question/(\d+))", [&ctrl](const Request& req, Response& res){
        std::string number = req.matches[1]; // **********************
        std::string html;
        ctrl.OneQuestion(number, &html);
        res.set_content(html, "text/html; charset=utf-8");
    });
    // 用户提交代码，使用我们的判题功能(1. 每道题的测试用例 2. compile_and_run)
    svr.Post(R"(/judge/(\d+))", [&ctrl](const Request& req, Response& res){
        std::string number = req.matches[1];
        std::string result_json;
        ctrl.Judge(number, req.body, &result_json);
        res.set_content(result_json, "application/json; charset=utf-8");
    });

    svr.set_base_dir("./wwwroot");
    svr.listen("0.0.0.0", 8080);
    return 0;
}